# 导入Flask框架
from flask import Flask, request, redirect, url_for, make_response, render_template


# 创建Flask应用实例
app = Flask(__name__)


# 定义用户资料页面路由，使用动态参数username
@app.route("/login", methods=["GET", "POST"])
def login():
    # 判断请求方法是否为POST
    if request.method == "POST":
        # 获取用户名和密码参数
        username = request.form.get("username")
        password = request.form.get("password")
        # 判断用户名和密码是否正确
        print(f"用户名: {username}, 密码: {password}")
        response = make_response("登录成功！")
        # 设置cookie
        response.headers["X-Custom-Header"] = "Value"
        return "登录成功！"
    else:
        # 请求方法为GET，显示登录页面
        return render_template("login.html", title="请登录")


@app.errorhandler(404)
def page_not_found(error):
    return render_template("404.html", error=error), 404


# 定义显示登录表单的函数，返回HTML格式的登录页面
def show_login_form():
    # 返回包含登录表单的完整HTML页面
    return """
    <!doctype html>
    <html>
    <head><title>登录</title></head>
    <body>
        <h1>用户登录</h1>
        <form method="POST">
            <p>用户名: <input type="text" name="username" required></p>
            <p>密码: <input type="password" name="password" required></p>
            <p><input type="submit" value="登录"></p>
        </form>
    </body>
    </html>
    """


# 判断是否直接运行此文件
if __name__ == "__main__":
    # 启动Flask开发服务器
    app.run()
